from enum import Enum

class TimeStrategy(Enum):
    CALL_AUCTION = 'CALL_AUCTION'  # 集合竞价
    TRADING = 'TRADING'  # 盘中
    AT_THE_CLOSE_ORDER = 'AT_THE_CLOSE_ORDER'  # 尾盘

class RecognitionStrategy(Enum):
    OPEN_HIGH = 'OPEN_HIGH'  # 高开
    HIGH_WALK = 'HIGH_WALK'  # 高走
    RED_THREE_SOLDIERS = 'RED_THREE_SOLDIERS'  # 红三兵
    BULLISH = 'BULLISH'  # 阳线（大涨后，大跌出现阳线）
    BULLISH_AFTER_MULTIPLE_BEARISH = 'BULLISH_AFTER_MULTIPLE_BEARISH'  # 2个及2个以上阴线后出现阳线
    LIMIT_UP_OF_TEN_DAYS = 'LIMIT_UP_OF_TEN_DAYS'  # 十日内涨停
    MULTIPLE_VOLUME_AND_PRICE_UP = 'MULTIPLE_VOLUME_AND_PRICE_UP'  # 多日量价齐升小阳线
    
    LONG_UPPER_SHADOW = 'LONG_UPPER_SHADOW'  # 长上影线-卖出预警
    MULTIPLE_BEARISH = 'MULTIPLE_BEARISH'  # 多连阴-卖出预警
    
    TURNOVER = 'TURNOVER'  # 成交量
    QUICKLY_PULL_UP = 'QUICKLYPULLUP'  # 快速拉升

RecognitionStrategyCN = {
    RecognitionStrategy.OPEN_HIGH.value: '高开',
    RecognitionStrategy.HIGH_WALK.value: '高走',
    RecognitionStrategy.RED_THREE_SOLDIERS.value: '红三兵',
    RecognitionStrategy.BULLISH.value: '阳线',
    RecognitionStrategy.BULLISH_AFTER_MULTIPLE_BEARISH.value: '多阴一阳',
    RecognitionStrategy.LIMIT_UP_OF_TEN_DAYS.value: '十日内涨停',
    RecognitionStrategy.MULTIPLE_VOLUME_AND_PRICE_UP.value: '多日量价齐升小阳线',
    
    RecognitionStrategy.LONG_UPPER_SHADOW.value: '长上影线',
    RecognitionStrategy.MULTIPLE_BEARISH.value: '多连阴',
    
    RecognitionStrategy.TURNOVER.value: '成交量',
    RecognitionStrategy.QUICKLY_PULL_UP.value: '快速拉升'
}

